/**********
This library is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the
Free Software Foundation; either version 3 of the License, or (at your
option) any later version. (See <http://www.gnu.org/copyleft/lesser.html>.)

This library is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for
more details.

You should have received a copy of the GNU Lesser General Public License
along with this library; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
**********/
// "liveMedia"
// Copyright (c) 1996-2018 Live Networks, Inc.  All rights reserved.
// AMR Audio File sinks
// Implementation

#include "include/AMRAudioFileSink.hh"
#include "include/AMRAudioSource.hh"
#include "include/OutputFile.hh"

////////// AMRAudioFileSink //////////

AMRAudioFileSink
::AMRAudioFileSink(UsageEnvironment &env, FILE *fid, unsigned bufferSize,
                   char const *perFrameFileNamePrefix)
        : FileSink(env, fid, bufferSize, perFrameFileNamePrefix),
          fHaveWrittenHeader(False) {
}

AMRAudioFileSink::~AMRAudioFileSink() {
}

AMRAudioFileSink *
AMRAudioFileSink::createNew(UsageEnvironment &env, char const *fileName,
                            unsigned bufferSize, Boolean oneFilePerFrame) {
    do {
        FILE *fid;
        char const *perFrameFileNamePrefix;
        if (oneFilePerFrame) {
            // Create the fid for each frame
            fid = NULL;
            perFrameFileNamePrefix = fileName;
        } else {
            // Normal case: create the fid once
            fid = OpenOutputFile(env, fileName);
            if (fid == NULL) break;
            perFrameFileNamePrefix = NULL;
        }

        return new AMRAudioFileSink(env, fid, bufferSize, perFrameFileNamePrefix);
    } while (0);

    return NULL;
}

Boolean AMRAudioFileSink::sourceIsCompatibleWithUs(MediaSource &source) {
    // The input source must be a AMR Audio source:
    return source.isAMRAudioSource();
}

void AMRAudioFileSink::afterGettingFrame(unsigned frameSize,
                                         unsigned numTruncatedBytes,
                                         struct timeval presentationTime) {
    AMRAudioSource *source = (AMRAudioSource *) fSource;
    if (source == NULL) return; // sanity check

    if (!fHaveWrittenHeader && fPerFrameFileNameBuffer == NULL) {
        // Output the appropriate AMR header to the start of the file.
        // This header is defined in RFC 4867, section 5.
        // (However, we don't do this if we're creating one file per frame.)
        char headerBuffer[100];
        sprintf(headerBuffer, "#!AMR%s%s\n",
                source->isWideband() ? "-WB" : "",
                source->numChannels() > 1 ? "_MC1.0" : "");
        unsigned headerLength = strlen(headerBuffer);
        if (source->numChannels() > 1) {
            // Also add a 32-bit channel description field:
            headerBuffer[headerLength++] = 0;
            headerBuffer[headerLength++] = 0;
            headerBuffer[headerLength++] = 0;
            headerBuffer[headerLength++] = source->numChannels();
        }

        addData((unsigned char *) headerBuffer, headerLength, presentationTime);
    }
    fHaveWrittenHeader = True;

    // Add the 1-byte header, before writing the file data proper:
    // (Again, we don't do this if we're creating one file per frame.)
    if (fPerFrameFileNameBuffer == NULL) {
        u_int8_t frameHeader = source->lastFrameHeader();
        addData(&frameHeader, 1, presentationTime);
    }

    // Call the parent class to complete the normal file write with the input data:
    FileSink::afterGettingFrame(frameSize, numTruncatedBytes, presentationTime);
}
